package com.xjgc.common.utils;

import com.google.common.collect.Lists;

import java.util.ArrayList;
import java.util.List;

public class CollectionUtils {

    /*
     * List分割
     */
    public static <T> List<List<T>> separateList(List<T> resList, int subListLength) {
        if (org.apache.commons.collections.CollectionUtils.isEmpty(resList) || subListLength <= 0) {
            return Lists.newArrayList();
        }
        List<List<T>> ret = Lists.newArrayList();
        int size = resList.size();
        if (size <= subListLength) {
            // 数据量不足 subListLength 指定的大小
            ret.add(resList);
        } else {
            int pre = size / subListLength;
            int last = size % subListLength;
            // 前面pre个集合，每个大小都是 subListLength 个元素
            for (int i = 0; i < pre; i++) {
                List<T> itemList = Lists.newArrayList();
                for (int j = 0; j < subListLength; j++) {
                    itemList.add(resList.get(i * subListLength + j));
                }
                ret.add(itemList);
            }
            // last的进行处理
            if (last > 0) {
                List<T> itemList = Lists.newArrayList();
                for (int i = 0; i < last; i++) {
                    itemList.add(resList.get(pre * subListLength + i));
                }
                ret.add(itemList);
            }
        }
        return ret;
    }

    public static void main(String[] args) {
        ArrayList<String> strings = new ArrayList<>();
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");
        strings.add("21312312");

        List<List<String>> lists = separateList(strings, 3);
        System.out.println(lists.size());
        for (List<String> list:lists) {
            for (String str:list){
                System.out.println(str);
            }
        }
    }

}
